// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#include <sstream>
#include <utility>

#include <aries/parquet/properties.h>

#include <nebula/io/buffered.h>
#include <nebula/io/memory.h>
#include <turbo/log/logging.h>
#include <nebula/future/thread_pool.h>

namespace parquet {
    std::shared_ptr<ArrowInputStream> ReaderProperties::GetStream(
        std::shared_ptr<ArrowInputFile> source, int64_t start, int64_t num_bytes) {
        if (buffered_stream_enabled_) {
            // ARROW-6180 / PARQUET-1636 Create isolated reader that references segment
            // of source
            PARQUET_ASSIGN_OR_THROW(
                std::shared_ptr<::nebula::io::InputStream> safe_stream,
                ::nebula::io::RandomAccessFile::get_stream(source, start, num_bytes));
            PARQUET_ASSIGN_OR_THROW(
                auto stream, ::nebula::io::BufferedInputStream::create(buffer_size_, pool_,
                    safe_stream, num_bytes));
            return stream;
        } else {
            PARQUET_ASSIGN_OR_THROW(auto data, source->read_at(start, num_bytes));

            if (data->size() != num_bytes) {
                std::stringstream ss;
                ss << "Tried reading " << num_bytes << " bytes starting at position " << start
                        << " from file but only got " << data->size();
                throw ParquetException(ss.str());
            }
            return std::make_shared<::nebula::io::BufferReader>(data);
        }
    }

    ::nebula::internal::Executor *ArrowWriterProperties::executor() const {
        return executor_ != nullptr ? executor_ : ::nebula::internal::get_cpu_thread_pool();
    }

    ArrowReaderProperties default_arrow_reader_properties() {
        static ArrowReaderProperties default_reader_props;
        return default_reader_props;
    }

    std::shared_ptr<ArrowWriterProperties> default_arrow_writer_properties() {
        static std::shared_ptr<ArrowWriterProperties> default_writer_properties =
                ArrowWriterProperties::Builder().build();
        return default_writer_properties;
    }
} // namespace parquet
